"""
-- 5 Задание. дана ведомость в которой два столбца - отдел и сотрудник.
-- необходимо подсчитать количество уникальных сотрудников в каждом отделе. результат отразить в поле Сnt
df: A, B
Отдел Сотрудник     Сnt
1     1             3
1     2             3
2     3             1
3     6             1
1     7             3
3     6             1
"""

import pandas as pd


def execute_task_5():
    data = [(1, 1), (1, 2), (2, 3), (3, 6), (1, 7), (3, 6)]
    df = pd.DataFrame(data, columns=['Отдел', 'Сотрудник'])
    df['Cnt'] = df.groupby('Отдел')['Сотрудник'].transform('nunique')
    print(df)


if __name__ == '__main__':
    execute_task_5()
